CUDA_VISIBLE_DEVICES='0,1' pythonc36 ../train.py \
--dataroot ../../GAN-RS/datasets/underwater/AB640 \
--name UW_pix2pix512 \
--suffix GRes9D45LS_A30U5G1 \
--model pix2pix \
--which_model_netG resnet_9blocks \
--which_direction AtoB \
--dataset_mode aligned \
--norm batch \
--pool_size 0 \
--checkpoints_dir ../checkpoints \
--save_epoch_freq 5 \
--batchSize 2 \
--display_winsize 512 \
--display_id 200 \
--display_freq 100 \
--print_freq 100 \
--lr_policy 'lambda' \
--lr 0.0002 \
--niter 50 \
--niter_decay 50 \
--gpu_ids '0,1' \
--loadSize 530 \
--fineSize 512 \
--which_model_netD 'multibranch' \
--n_layers_D 4 \
--n_layers_U 5 \
--lambda_A 30 \
--lambda_U 5 \
--lambda_GAN 1
#'resnet_9blocks' \